package classic

/**
 * 给你一个 升序排列 的数组 nums ，请你 原地 删除重复出现的元素，使每个元素 只出现一次 ，
 * 返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。然后返回 nums 中唯一元素的个数。
 *
 * 考虑 nums 的唯一元素的数量为 k ，你需要做以下事情确保你的题解可以被通过：
 *
 * 更改数组 nums ，使 nums 的前 k 个元素包含唯一元素，并按照它们最初在 nums 中出现的顺序排列。nums 的其余元素与 nums 的大小不重要。
 * 返回 k 。
 *
 * 判题标准:
 *
 * 系统会用下面的代码来测试你的题解:
 *
 * int[] nums = [...]; // 输入数组
 * int[] expectedNums = [...]; // 长度正确的期望答案
 *
 * int k = removeDuplicates(nums); // 调用
 *
 * assert k == expectedNums.length;
 * for (int i = 0; i < k; i++) {
 *     assert nums[i] == expectedNums[i];
 * }
 * 如果所有断言都通过，那么您的题解将被 通过。
 *
 * 示例 1：
 *
 * 输入：nums = [1,1,2]
 * 输出：2, nums = [1,2,_]
 * 解释：函数应该返回新的长度 2 ，并且原数组 nums 的前两个元素被修改为 1, 2 。不需要考虑数组中超出新长度后面的元素。
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/remove-duplicates-from-sorted-array
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
fun removeDuplicates(nums: IntArray): Int {
    val len = nums.size
    if (len < 2) return len
    var fastIndex = 1
    var slowIndex = 0
    while (fastIndex < len) {
        if (nums[fastIndex] != nums[slowIndex]) {
            nums[++slowIndex] = nums[fastIndex]
        }
        ++fastIndex
    }
    return slowIndex + 1
}

/**
 * 给你一个有序数组 nums ，请你 原地 删除重复出现的元素，使得出现次数超过两次的元素只出现两次 ，返回删除后数组的新长度。
 *
 * 不要使用额外的数组空间，你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
 *
 * 示例1：
 * 输入：nums = [1,1,1,1,2,2,3]
 * 输出：5, nums = [1,1,2,2,3]
 * 解释：函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3 。 不需要考虑数组中超出新长度后面的元素。
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/remove-duplicates-from-sorted-array-ii
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
fun removeDuplicates2(nums: IntArray): Int {
    val len = nums.size
    if (len < 3) return len
    var fastIndex = 1
    var slowIndex = 0
    var mark = 1
    while (fastIndex < len) {
        if (nums[fastIndex] == nums[slowIndex]) {
            ++mark
        } else {
            mark = 1
        }
        if (mark < 3) {
            ++slowIndex
            nums[slowIndex] = nums[fastIndex]
        }
        ++fastIndex
    }
    return slowIndex + 1
}

fun main() {
    val arrays = intArrayOf(1, 1, 1, 1, 2, 2, 3, 4, 4, 4, 4, 5, 5, 6)
    println(arrays.contentToString())
    println(removeDuplicates2(arrays))
    println(arrays.contentToString())
}
